import unittest

from api.ihrm_emp import IHRMEmpApi
from api.ihrm_login_api import IHRMLoginApi
from utils.assert_utils import IHRMAssertUtils

import random


class TestIHRMAddEmp(unittest.TestCase):
    def test_add_emp_ok(self):
        resp = IHRMLoginApi.login_request({
            'mobile': '13800000002',
            'password': '123456'
        })
        token = resp.json().get('data')
        ran_mobile = random.randint(10000000, 99999999)
        resp = IHRMEmpApi.add_emp(token, {
            'mobile': f'138{ran_mobile}',
            'username': '大黄',
            'workNumber': '5678'
        })
        IHRMAssertUtils.assert_add_emp(self, resp, True, 10000, '操作成功！')

        emp_id = resp.json().get('data').get('id')
        # 调用删除员工的接口 不过后端使用的是逻辑删除, 所以不能使用这种方式
        # resp = IHRMEmpApi.del_emp(token, emp_id)
        # 直接操作数据库, 删除这条数据
        # DBUtils.uid_db(f'delete from 员工表 where id={emp_id}')

    def test_add_emp_mobile_error(self):
        pass

    def test_add_emp_worknum_error(self):
        pass
